package com.atguigu.tingshu.common.login;

import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.execption.GuiguException;
import com.atguigu.tingshu.common.result.ResultCodeEnum;
import com.atguigu.tingshu.common.util.AuthContextHolder;
import com.atguigu.tingshu.vo.user.UserInfoVo;
import jakarta.servlet.http.HttpServletRequest;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * ClassName: LoginAspect
 * Package: com.atguigu.tingshu.common.login
 * Description:
 * 切面：将通知作用到切入点上
 * 对所有业务模块的controller层中使用了@GuiguLogin的注解方法进行增强。
 *
 * @Author 咚咚小圆帽
 * @Create 2025/6/29 14:19
 * @Version 1.0
 */
@Aspect
@Component
@Slf4j
public class LoginAspect {

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 环绕通知
     * 切入点表达式：第一个*代表任意返回类型，第二个*代表包名，
     * 第三个*代表类名，第四个*代表方法名, ..代表任意参数
     *
     * @param pjp        目标方法对象
     * @param guiguLogin 注解对象
     * @return
     * @throws Throwable
     */
    @Around("execution(* com.atguigu.tingshu.*.api.*.*(..)) && @annotation(guiguLogin)")
    public Object doBasicProfiling(ProceedingJoinPoint pjp, GuiguLogin guiguLogin) throws Throwable {
        //一.前置通知
        //1.获取请求头中的token令牌值
        //1.1 获取请求对象-获取上层接口
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        //1.2 强制转换为ServletRequestAttributes类对象
        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) requestAttributes;
        HttpServletRequest request = servletRequestAttributes.getRequest();
        //1.3 获取token令牌值
        String token = request.getHeader("token");

        //2.根据token查询Redis中的用户信息
        String loginKey = RedisConstant.USER_LOGIN_KEY_PREFIX + token;
        UserInfoVo userInfoVo = (UserInfoVo) redisTemplate.opsForValue().get(loginKey);

        //3.如果目标方法注解的required的属性为true，并且用户信息为空，则抛出异常，
        // 并且设置业务状态码为208，前端捕获异常，跳转到登录界面。
        if (guiguLogin.required() && userInfoVo == null) {
            throw new GuiguException(ResultCodeEnum.LOGIN_AUTH);
        }

        //4.如果用户信息不为空，则将用户id存入到ThreadLocal中。
        if (userInfoVo != null) {
            Long userId = userInfoVo.getId();
            AuthContextHolder.setUserId(userId);
        }


        //二.执行目标方法
        Object val = pjp.proceed();

        //三.后置通知
        //5.避免ThreadLocal内存泄露导致OOM问题，手动删除ThreadLocal中的数据
        AuthContextHolder.removeUserId();

        return val;
    }

}
